# Path config

from pathlib import Path
BASE_DIR = Path('common')
paths = {
    'data_dir': BASE_DIR / 'data',
    'train_data': BASE_DIR / 'data' / 'datagrand_2021_train.csv',
    'test_data': BASE_DIR / 'data' / 'datagrand_2021_test.csv',
    'pretrain_data':  BASE_DIR / 'data' / 'datagrand_2021_unlabeled_data.json',
    'sample':  BASE_DIR / 'data' / 'sample.json',
    'log_dir': BASE_DIR / 'output' / 'log' / 'log.log',
    'output': BASE_DIR / 'output',
    'model_path': BASE_DIR / 'models',
    # 'writer_dir': BASE_DIR / "output/TSboard",
    # 'figure_dir': BASE_DIR / "output/figure",
    # 'checkpoint_dir': BASE_DIR / "output/checkpoints",
    # 'cache_dir': BASE_DIR / 'model/',
    # 'result_dir': BASE_DIR / "output/result",
}
